Repeating a program using the For loop in python